Program Listing for File global_config.py

↰ Return to documentation for file (codes/lanekerbnetros/config/global_config.py)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Time    : 17-05-2019
# @Author  : Zhou Hui
# @Original site    : https://github.com/MaybeShewill-CV/lanenet-lane-detection
# @File    : lanenet_node.py
"""
Config the global variable
"""
from easydict import EasyDict as edict

__C = edict()


cfg = __C


__C.TRAIN = edict()


__C.TRAIN.EPOCHS = 200010

__C.TRAIN.DISPLAY_STEP = 1

__C.TRAIN.TEST_DISPLAY_STEP = 1000

__C.TRAIN.MOMENTUM = 0.9

__C.TRAIN.LEARNING_RATE = 0.0005

__C.TRAIN.GPU_MEMORY_FRACTION = 0.85

__C.TRAIN.TF_ALLOW_GROWTH = True

__C.TRAIN.BATCH_SIZE = 8


__C.TRAIN.VAL_BATCH_SIZE = 8

__C.TRAIN.LR_DECAY_STEPS = 410000

__C.TRAIN.LR_DECAY_RATE = 0.1

__C.TRAIN.CLASSES_NUMS = 2

__C.TRAIN.IMG_HEIGHT = 256

__C.TRAIN.IMG_WIDTH = 512


__C.TEST = edict()


__C.TEST.GPU_MEMORY_FRACTION = 0.5

__C.TEST.TF_ALLOW_GROWTH = True

__C.TEST.BATCH_SIZE = 8